



###
 # @Author: daniel
 # @Date: 2023-10-15 21:27:32
 # @LastEditTime: 2023-10-17 14:50:59
 # @LastEditors: daniel
 # @Description: 
 # @FilePath: /MVT-3DVG/train.sh
 # have a nice day
### 
source /usr/local/miniconda3/etc/profile.d/conda.sh 
conda activate cerberus2


PATH_OF_SR3D_DATA='';
PATH_OF_SCANNET_FILE=/home/DISCOVER_summer2022/xusc/exp/co-t3-dref/refering_codes/MVT-3DVG/data/preprocessed_referit3d_scans/keep_all_points_with_global_scan_alignment/keep_all_points_with_global_scan_alignment.pkl;
PATH_OF_REFERIT3D_FILE=/home/DISCOVER_summer2022/xusc/exp/co-t3-dref/refering_codes/MVT-3DVG/data/refer_it_3d/sr3d.csv;
PATH_OF_BERT='/home/DISCOVER_summer2022/xusc/exp/co-t3-dref/refering_codes/MVT-3DVG/bert_model/pytorch_model.bin';

gpuids='6';

CUDA_VISIBLE_DEVICES=$gpuids python train_referit3d.py \
   -scannet-file $PATH_OF_SCANNET_FILE$\
    -referit3D-file $PATH_OF_REFERIT3D_FILE$ \
    --bert-pretrain-path $PATH_OF_BERT$ \
    --log-dir logs/COT3DRef_sr3d \
    --n-workers 16 \
    --model 'referIt3DNet_transformer' \
    --unit-sphere-norm True \
    --batch-size 24 \
    --encoder-layer-num 3 \
    --decoder-layer-num 4 \
    --decoder-nhead-num 8 \
    --gpu "0" \
    --view_number 4 \
    --rotate_number 4 \
    --label-lang-sup True \
    --anchors "cot" \
    --max-test-objects 52 \
    --cot_type "cross" \
    --predict_lang_anchors True \
    --lang_filter_objs False \
    --visaug_shuffle_mode 'none' \
    --visaug_extracted_obj_path '$PATH_OF_SR3D_DATA' \
    --visaug_pc_augment True \
    --train_data_percent 1.0